python train_mnist.py -c ../../config/mnist.yaml